package handler

import (
	"fmt"
	"multi-cloud-manage/config"
	"multi-cloud-manage/internal/model"
	"multi-cloud-manage/internal/storage"
	"multi-cloud-manage/utils"
	"time"

	"github.com/gin-gonic/gin"
)

// 注册路由函数
func Register(c *gin.Context) {
	// 获取用户名和密码(POST表单)
	username := c.PostForm("username")
	password := c.PostForm("password")
	// 判断用户或者密码是否为空
	if username == "" || password == "" {
		// 如果是前后端分离项目
		c.JSON(200, gin.H{
			"code":    400,
			"message": "用户名或者密码不能为空",
		})
		return
	}
	// 密码加密 bcrypt
	hashPwd, err := utils.HashPassword(password)
	if err != nil {
		c.JSON(200, gin.H{
			"code":    401,
			"message": "加密失败",
		})
		return
	}
	fmt.Println(hashPwd)
	// 检测用户是否存在
	_, exists := storage.GetUserByUsername(username)
	if exists {
		c.JSON(200, gin.H{
			"code":    402,
			"message": "该用户已被注册",
		})
		return
	}
	// 将数据插入到数据库
	user := model.User{
		Username: username,
		Password: hashPwd,
		CreateAt: time.Now(),
	}
	storage.CreateUser(user)
	c.JSON(200, gin.H{
		"code":    200,
		"message": "注册成功",
	})
}

// 登录
func Login(c *gin.Context) {
	// 获取表单传参中的用户名和密码
	username := c.PostForm("username")
	password := c.PostForm("password")
	if username == "" || password == "" {
		c.JSON(200, gin.H{
			"code":    401,
			"message": "用户名或密码不能为空",
		})
		return
	}
	// 查询用户
	user, exists := storage.GetUserByUsername(username)
	if !exists { // 如果不存在
		c.JSON(200, gin.H{
			"code":    402,
			"message": "该用户不存在",
		})
		return
	}
	// 验证密码（如果不正确）
	if !utils.CheckPasswordHash(password, user.Password) {
		c.JSON(200, gin.H{
			"code":    403,
			"message": "密码错误",
		})
		return
	}
	jwt_secret := config.GetConfig().Jwt.Secret
	// 生成token
	token, err := utils.GenerateJWT(user, jwt_secret)
	fmt.Println("秘钥：", jwt_secret)
	if err != nil {
		c.JSON(200, gin.H{
			"code":    404,
			"message": "token生成失败",
		})
		fmt.Println(err)
		return
	}
	c.JSON(200, gin.H{
		"code":    200,
		"message": "登录成功",
		"token":   token,
	})
}
